package com.game.utils;

import com.google.common.base.Charsets;
import com.google.common.io.Files;

import java.io.*;
import java.nio.charset.StandardCharsets;
import java.text.DecimalFormat;
import java.util.List;
import java.util.Optional;

/**
 * 文件操作工具类.
 *
 * @author wh
 */
public class FileUtils {
    /**
     * 可读大小的单位
     */
    private static final String[] UNITS = new String[]{"B", "KB", "MB", "GB", "TB", "EB"};

    /**
     * 加载类路径下指定名称文件中的文本.
     *
     * @param fileName 文件名称
     * @return 返回文件中的文本
     */
    public static Optional<String> loadFileText(String fileName) {
        try (InputStream is = Thread.currentThread().getContextClassLoader().getResourceAsStream(fileName)) {
            return Optional.ofNullable(StringUtils.readString(is));
        } catch (Exception e) {
        }
        // 文件不存在等其他情况返回null
        return Optional.empty();
    }

    /**
     * 读取指定名称文件中的文本.
     *
     * @param fileName 文件名称
     * @return 返回文件中的文本
     * @throws IOException           If an I/O error occurs
     * @throws FileNotFoundException 文件未找到会抛出此异常
     */
    public static String readFileText(String fileName) throws FileNotFoundException, IOException {
        try (FileReader reader = new FileReader(fileName)) {
            return StringUtils.readString(reader);
        }
    }

    /**
     * 写入指定文本到文件中.
     * <p>
     * 文件不存在，则会自动创建，默认是覆盖原文件
     *
     * @param fileName 文件名称
     * @param content  要写入的内容
     * @throws IOException If an I/O error occurs
     */
    public static void writeFileText(String fileName, String content) throws IOException {
        writeFileText(fileName, false, content);
    }

    /**
     * 写入指定文本到文件中.
     * <p>
     * 文件不存在，则会自动创建
     *
     * @param fileName 文件名称
     * @param append   是否追加写入
     * @param content  要写入的内容
     * @throws IOException If an I/O error occurs
     */
    public static void writeFileText(String fileName, boolean append, String content) throws IOException {
        try (FileOutputStream fos = new FileOutputStream(fileName, append);
             OutputStreamWriter osw = new OutputStreamWriter(fos, Charsets.UTF_8)) {
            osw.write(content);
            osw.flush();
        }
    }


    public static void writerFile(String fileName, byte[] data, boolean append) throws IOException {
        File file = new File(fileName);
        createNewFile(file);
        try (FileOutputStream fos = new FileOutputStream(file, append);
             OutputStream out = new BufferedOutputStream(fos)) {
            out.write(data);
            out.flush();
        }
    }

    /**
     * 可读的文件大小
     *
     * @param file 文件
     * @return 大小
     */
    public static String readableFileSize(File file) {
        return readableFileSize(file.length());
    }

    /**
     * 可读的文件大小<br>
     *
     * @param size Long类型大小
     * @return 大小
     */
    public static String readableFileSize(long size) {
        if (size <= 0) {
            return "0";
        }
        int digitGroups = (int) (Math.log10(size) / Math.log10(1024));
        return new DecimalFormat("#,##0.##").format(size / Math.pow(1024, digitGroups)) + " " + UNITS[digitGroups];
    }

    /**
     * 创建指定文件,目录不存在则自动创建
     * <p>
     * 如果目标文件不存在并且创建成功，则为true；如果目标文件存在，则为false
     *
     * @param file 文件对象
     * @return 如果目标文件不存在并且创建成功，则为true；如果目标文件存在，则为false
     * @throws IOException IO异常
     */
    public static boolean createNewFile(File file) throws IOException {
        // 目标文件存在，直接返回false.
        if (file.exists()) {
            return false;
        }

        // 如果目录不存在则创建此父目录
        final File parentDir = file.getParentFile();
        if (parentDir != null && !parentDir.exists()) {
            parentDir.mkdirs();
        }

        // 目录有了，直接调用JDK的创建新文件命令
        return file.createNewFile();
    }

    /**
     * 按行读取文件
     *
     * @param fileName
     * @return
     */
    public static List<String> readFileByLine(String fileName) {
        File file = new File(fileName);
        if (file == null || !file.exists())
            return null;
        try {
            return Files.readLines(file, StandardCharsets.UTF_8);
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * 读取文件
     *
     * @param file
     * @param encoding
     * @return
     */
    public static String read(File file, String encoding) {
        String string = "";
        try {
            BufferedReader in = new BufferedReader(new InputStreamReader(new FileInputStream(file), encoding));
            String str = "";
            while ((str = in.readLine()) != null) {
                string += str + "\r\n";
            }
            in.close();
        } catch (Exception ex) {
            ex.printStackTrace();
        }
        return string;
    }
}